package com.itheima;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;

/**
 * @author Xiao
 * @date 2021/9/19
 */
public class Generator {
    public static void main(String[] args) {
        //1.创建代码生成器对象，执行生成代码操作
        AutoGenerator generator = new AutoGenerator();

        //2.设置数据源相关配置，读取数据库信息，根据数据库表结构自动生成代码
        DataSourceConfig dataSource = new DataSourceConfig();
        dataSource.setDriverName("com.mysql.cj.jdbc.Driver");
        dataSource.setUrl("jdbc:mysql://localhost:3306/db10?useSSL=false&useServerPrepStmts=true");
        dataSource.setUsername("root");
        dataSource.setPassword("1234");
        generator.setDataSource(dataSource);

        //设置全局配置
        GlobalConfig globalConfig = new GlobalConfig();
        //设置代码生成的位置 --> user.dir：用户所在目录     "/..."：项目下的路径
        globalConfig.setOutputDir(System.getProperty("user.dir") +
                "/mybatisplus_generator/src/main/java");
        globalConfig.setOpen(false);    //设置生成完毕后是否打开代码所在目录
        globalConfig.setAuthor("coderXiao");    //设置作者
        globalConfig.setFileOverride(true);    //设置是否覆盖原始生成的文件
        globalConfig.setMapperName("%sDao");    //设置数据层接口名，%s为占位符，指代模块名称
        globalConfig.setIdType(IdType.ASSIGN_ID);   //设置id生成策略
        generator.setGlobalConfig(globalConfig);

        //设置包名相关配置
        PackageConfig packageConfig = new PackageConfig();
        //设置生成的包名，与代码所在位置不冲突，二者叠加组成完整路径
        packageConfig.setParent("com.itheima");
        packageConfig.setEntity("com.itheima.domain");  //设置实体类的包名
        packageConfig.setMapper("dao"); //设置数据层的包名
        generator.setPackageInfo(packageConfig);

        //策略设置
        StrategyConfig strategyConfig = new StrategyConfig();
        strategyConfig.setInclude("tbl_user");   //设置当前参与生成的表明，参数为可变参数
        //设置数据库表的前缀名称，实体类名=数据库表名-前缀名，如 User = tb_user - tb_
        strategyConfig.setTablePrefix("tbl_");
        strategyConfig.setRestControllerStyle(true);    //设置是否启用Rest风格
        strategyConfig.setVersionFieldName("version");  //设置乐观锁字段名
        strategyConfig.setLogicDeleteFieldName("deleted");  //设置逻辑删除字段名
        strategyConfig.setEntityLombokModel(true);  //设置是否启用lombok
        generator.setStrategy(strategyConfig);

        //4.执行自动生成操作
        generator.execute();
    }
}
